# from sklearn.feature_extraction.text import TfidfVectorizer
#
# from dnn.recall.Bm25Vectorizer import Bm25Vectorizer
#
#
# tfidf_vec = TfidfVectorizer()
#
# data = [
#     'hello hello world',
#     'oh hello there',
#     'Play it',
#     'Play it again Sam,24343,123',
# ]
#
# ret1 = tfidf_vec.fit_transform(data)
# ret2 = Bm25Vectorizer().fit_transform(data)
#
# print(ret1.toarray())
# print('*' * 50)
# print(ret2)
import torch

print(torch.cuda.is_available())

